package com.linyaonan.leetcode.medium._129;

import java.util.ArrayList;
import java.util.List;

/**
 *
 * 给定一个二叉树，它的每个结点都存放一个 0-9 的数字，每条从根到叶子节点的路径都代表一个数字。
 *
 * 例如，从根到叶子节点路径 1->2->3 代表数字 123。
 *
 * 计算从根到叶子节点生成的所有数字之和。
 *
 * 说明: 叶子节点是指没有子节点的节点。
 *
 * 示例 1:
 *
 * 输入: [1,2,3]
 *     1
 *    / \
 *   2   3
 * 输出: 25
 * 解释:
 * 从根到叶子节点路径 1->2 代表数字 12.
 * 从根到叶子节点路径 1->3 代表数字 13.
 * 因此，数字总和 = 12 + 13 = 25.
 * 示例 2:
 *
 * 输入: [4,9,0,5,1]
 *     4
 *    / \
 *   9   0
 *  / \
 * 5   1
 * 输出: 1026
 * 解释:
 * 从根到叶子节点路径 4->9->5 代表数字 495.
 * 从根到叶子节点路径 4->9->1 代表数字 491.
 * 从根到叶子节点路径 4->0 代表数字 40.
 * 因此，数字总和 = 495 + 491 + 40 = 1026.
 *
 *
 * @author: Lin
 * @date: 2020/1/18
 */
public class SumRootNumbers {
    List<String> save;
    public int sumNumbers(TreeNode root) {
        save = new ArrayList<>();
        StringBuilder sb = new StringBuilder();
        dfs(root, sb);
        int sum = 0;
        for (String s : save) {
            sum += Integer.valueOf(s);
        }
        return sum;
    }

    private void dfs(TreeNode node, StringBuilder sb) {
        if (node != null) {
            sb.append(node.val);
            if (node.left == null && node.right == null) {
                String s = sb.toString();
                save.add(s);
            } else {
                if (node.left != null) {
                    dfs(node.left, sb);
                    sb.delete(sb.length() - 1, sb.length());
                }
                if (node.right != null) {
                    dfs(node.right, sb);
                    sb.delete(sb.length() - 1, sb.length());
                }
            }
        }
    }
}
